op {
  graph_op_name: "BatchMatrixDiagPart"
}
